package leetcode;

/*
290. 单词规律
给定一种规律 pattern 和一个字符串 str ，判断 str 是否遵循相同的规律。
这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。
示例1:
输入: pattern = "abba", str = "dog cat cat dog"
输出: true
示例 2:
输入:pattern = "abba", str = "dog cat cat fish"
输出: false
示例 3:
输入: pattern = "aaaa", str = "dog cat cat dog"
输出: false
示例 4:
输入: pattern = "abba", str = "dog dog dog dog"
输出: false
说明:
你可以假设 pattern 只包含小写字母， str 包含了由单个空格分隔的小写字母。
*/

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Stack;

public class problems_290 {
    public static void main(String[] args) {
        System.out.println(new Solution().wordPattern("abba", "dog cat cat dog"));
    }

    static class Solution {
        public boolean wordPattern(String pattern, String str) {
            String[] split = str.split(" ");
            if (pattern.length() != split.length) return false;
            // 比对
            Map<Character, String> mapChar = new HashMap<>();
            Map<String, Character> mapStr = new HashMap<>();
            for (int i = 0; i < pattern.length(); i++) {
                Character key = pattern.charAt(i);
                String strSub = split[i];
                if(mapChar.containsKey(key)){
                    if(!mapChar.get(key).equals(strSub)) return false;
                }else {
                    if(mapChar.containsKey(strSub)) return false;
                    mapChar.put(key, strSub);
                }
                if(mapStr.containsKey(strSub)){
                    if(!mapStr.get(strSub).equals(key)) return false;
                }else {
                    mapStr.put(strSub, key);
                }
            }
            return true;
        }
    }
}